from langfuse.openai import openai  # OpenAI integration
from langfuse.decorators import observe

# 加载 .env 到环境变量
from dotenv import load_dotenv, find_dotenv

_ = load_dotenv(find_dotenv())


# 在langfuse中，如果没有定义名称，名称默认就使用方法的名称，比如下面的就是run
@observe()
def run():
    return openai.chat.completions.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "user", "content": "对我说Hello, World!"}
        ],
    ).choices[0].message.content


print(run())
